package com.speech.utils;

import android.content.Context;
import android.os.Build;

import me.leolin.shortcutbadger.ShortcutBadger;

public class BadgerUtil {

    public static void setBadger(Context context, Integer iMessageNum) {
        if (Build.MANUFACTURER.equalsIgnoreCase("Xiaomi")) {
            //TODO Badger
//            ShortcutBadger.applyNotification(context,,iMessageNum);
        } else {
            ShortcutBadger.applyCount(context, iMessageNum); //for 1.1.4+
        }
    }

    public static void removeBadger(Context context) {
        ShortcutBadger.removeCount(context); //for 1.1.4+
    }

}
